#!/bin/bash

#!/bin/bash

export WANDB_MODE=disabled

# Optionally set the cache for transformers
# export TRANSFORMERS_CACHE='YOUR_PATH/huggingface'

export model=$1 # llama2 or vicuna
export setup=$2 # behaviors or strings

# Create results folder if it doesn't exist
if [ ! -d "../results" ]; then
    mkdir "../results"
    echo "Folder '../results' created."
else
    echo "Folder '../results' already exists."
fi

for data_offset in 0 10 20 30 40 50 60 70 80 90
do

    # python -u ../main.py \
    #     --config="../configs/individual_${model}.py" \
    #     --config.attack=gcg \
    #     --config.train_data="../../data/advbench/harmful_${setup}.csv" \
    #     --config.result_prefix="../../../../results/individual_${setup}_${model}_gcg_offset${data_offset}" \
    #     --config.n_train_data=10 \
    #     --config.data_offset=$data_offset \
    #     --config.n_steps=1000 \
    #     --config.test_steps=50 \
    #     --config.batch_size=512
        python -u main.py \
        --config="./configs/individual_llama2.py" \
        --config.attack=gcg \
        --config.train_data="../data/advbench/harmful_behaviors.csv" \
        --config.result_prefix="../../../individual" \
        --config.n_train_data=100 \
        --config.n_steps=30 \
        --config.test_steps=5 \
        --config.batch_size=64

done